# -*- coding: utf-8 -*-
"""
测试 llm_provider.chat 函数的简单脚本
"""

from llm_provider import chat

def test_chat():
    """测试 chat 函数的基本功能"""
    
    # 获取用户输入
    user_input = input("请输入您的测试消息: ")
    messages = [
        {"role": "user", "content": user_input}
    ]
    
    print("测试 chat 函数:")
    print("-" * 40)
    
    # 调用 chat 函数
    try:
        response = chat(messages)
        print("输入消息:")
        for msg in messages:
            print(f"  {msg['role']}: {msg['content']}")
        print("\n返回内容:")
        print(response)
        print("-" * 40)
        
    except Exception as e:
        print(f"调用 chat 函数时出错: {e}")

def test_chat_with_json_mode():
    """测试 chat 函数的 JSON 模式"""
    
    # 获取用户输入
    user_input = input("请输入您的测试消息(要求返回JSON格式): ")
    messages = [
        {"role": "user", "content": user_input}
    ]
    
    print("\n测试 chat 函数 (JSON 模式):")
    print("-" * 40)
    
    # 调用 chat 函数，启用 JSON 模式
    try:
        response = chat(messages, json_mode=True)
        print("输入消息:")
        for msg in messages:
            print(f"  {msg['role']}: {msg['content']}")
        print("\n返回内容:")
        print(response)
        print("-" * 40)
        
    except Exception as e:
        print(f"调用 chat 函数时出错: {e}")

if __name__ == "__main__":
    test_chat()